package leetcode_题库;

/**
 * Created by IntelliJ IDEA.
 *
 * @ProjectName leetcode
 * @Author Ding
 * @CreateTime 2022/8/29 9:49 星期一
 */
public class _700_二叉搜索树中的搜索 {
    /**
     * 迭代：
     * > 2022/08/29 10:00:34
     * 解答成功:
     * 	执行耗时:0 ms,击败了100.00% 的Java用户
     * 	内存消耗:42.1 MB,击败了25.02% 的Java用户
     * 递归：
     * > 2022/08/29 10:03:06
     * 解答成功:
     * 	执行耗时:0 ms,击败了100.00% 的Java用户
     * 	内存消耗:42.2 MB,击败了9.18% 的Java用户
     * @param root
     * @param val
     * @return
     */
    public TreeNode searchBST(TreeNode root, int val) {
        //迭代
        while (root != null && root.val != val) {
            root = root.val > val ? root.left : root.right;
        }
        return root;
        //递归
//        return dfs(root,val);
    }

    public TreeNode dfs(TreeNode n, int val) {
        if (n == null || n.val == val) {
            //无论n为null或找到n，都返回n即可
            return n;
        }
        //否则，由于二叉树左小右大，可以三元表达式选择递归
        return n.val > val ?  dfs(n.left, val) : dfs(n.right, val);
    }

    public class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;

        TreeNode() {
        }

        TreeNode(int val) {
            this.val = val;
        }

        TreeNode(int val, TreeNode left, TreeNode right) {
            this.val = val;
            this.left = left;
            this.right = right;
        }
    }
}
